#ifndef CONV_H
#define CONV_H

#include "tensor.h"

// 卷积相关的参数结构体，包含卷积常用的参数信息
struct ConvolutionParams {
    int stride;
    int padding;
};

Tensor convolution_forward_gpu(Tensor input, Tensor kernel, ConvolutionParams params);
Tensor convolution_forward_cpu(Tensor input, Tensor kernel, ConvolutionParams params);
#endif // CONV_H
